// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_INDEXED_DB_DATABASE_IMPL_H_
#define CONTENT_BROWSER_INDEXED_DB_DATABASE_IMPL_H_

#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "content/common/indexed_db/indexed_db.mojom.h"

namespace content {

class IndexedDBConnection;
class IndexedDBDispatcherHost;

class DatabaseImpl : public ::indexed_db::mojom::Database {
public:
    explicit DatabaseImpl(std::unique_ptr<IndexedDBConnection> connection,
        const url::Origin& origin,
        scoped_refptr<IndexedDBDispatcherHost> dispatcher_host);
    ~DatabaseImpl() override;

    // ::indexed_db::mojom::Database implementation
    void CreateObjectStore(int64_t transaction_id,
        int64_t object_store_id,
        const base::string16& name,
        const IndexedDBKeyPath& key_path,
        bool auto_increment) override;
    void DeleteObjectStore(int64_t transaction_id,
        int64_t object_store_id) override;
    void RenameObjectStore(int64_t transaction_id,
        int64_t object_store_id,
        const base::string16& new_name) override;
    void CreateTransaction(int64_t transaction_id,
        const std::vector<int64_t>& object_store_ids,
        blink::WebIDBTransactionMode mode) override;
    void Close() override;
    void VersionChangeIgnored() override;
    void AddObserver(int64_t transaction_id,
        int32_t observer_id,
        bool include_transaction,
        bool no_records,
        bool values,
        uint16_t operation_types) override;
    void RemoveObservers(const std::vector<int32_t>& observers) override;
    void Get(int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const IndexedDBKeyRange& key_range,
        bool key_only,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void GetAll(
        int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const IndexedDBKeyRange& key_range,
        bool key_only,
        int64_t max_count,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void Put(int64_t transaction_id,
        int64_t object_store_id,
        indexed_db::mojom::ValuePtr value,
        const IndexedDBKey& key,
        blink::WebIDBPutMode mode,
        const std::vector<IndexedDBIndexKeys>& index_keys,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void SetIndexKeys(int64_t transaction_id,
        int64_t object_store_id,
        const IndexedDBKey& primary_key,
        const std::vector<IndexedDBIndexKeys>& index_keys) override;
    void SetIndexesReady(int64_t transaction_id,
        int64_t object_store_id,
        const std::vector<int64_t>& index_ids) override;
    void OpenCursor(
        int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const IndexedDBKeyRange& key_range,
        blink::WebIDBCursorDirection direction,
        bool key_only,
        blink::WebIDBTaskType task_type,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks_info) override;
    void Count(
        int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const IndexedDBKeyRange& key_range,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void DeleteRange(
        int64_t transaction_id,
        int64_t object_store_id,
        const IndexedDBKeyRange& key_range,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void Clear(
        int64_t transaction_id,
        int64_t object_store_id,
        ::indexed_db::mojom::CallbacksAssociatedPtrInfo callbacks) override;
    void CreateIndex(int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const base::string16& name,
        const IndexedDBKeyPath& key_path,
        bool unique,
        bool multi_entry) override;
    void DeleteIndex(int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id) override;
    void RenameIndex(int64_t transaction_id,
        int64_t object_store_id,
        int64_t index_id,
        const base::string16& new_name) override;
    void Abort(int64_t transaction_id) override;
    void Commit(int64_t transaction_id) override;
    void AckReceivedBlobs(const std::vector<std::string>& uuids) override;

private:
    class IDBThreadHelper;

    IDBThreadHelper* helper_;
    scoped_refptr<IndexedDBDispatcherHost> dispatcher_host_;
    const url::Origin origin_;
    scoped_refptr<base::SingleThreadTaskRunner> idb_runner_;

    DISALLOW_COPY_AND_ASSIGN(DatabaseImpl);
};

} // namespace content

#endif // CONTENT_BROWSER_INDEXED_DB_DATABASE_IMPL_H_
